package cn.xkai.exercise.b;

import com.google.gson.Gson;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @description: 两个列表的最小索引总和
 * 假设 Andy 和 Doris 想在晚餐时选择一家餐厅，并且他们都有一个表示最喜爱餐厅的列表，每个餐厅的名字用字符串表示。
 * 你需要帮助他们用最少的索引和找出他们共同喜爱的餐厅。 如果答案不止一个，则输出所有答案并且不考虑顺序。 你可以假设答案总是存在。
 * <p>
 * 自己的思路：把长数组数据存放到哈希表中，通过短数组遍历计算索引和，根据最小索引更新StringBuilder，然后根据逗号分组
 * 借鉴的思路：
 * @author: kaixiang
 * @date: 2022-07-27
 **/
public class Solution100 {
    public String[] findRestaurant(String[] list1, String[] list2) {
        return list1.length > list2.length ? getMin(list1, list2) : getMin(list2, list1);
    }

    private String[] getMin(String[] l1, String[] l2) {
        int min = 2 * l1.length, head = 0;
        HashMap<String, Integer> map = new HashMap<>(l1.length);
        for (int i = 0; i < l1.length; i++) {
            map.put(l1[i], i);
        }
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < l2.length; i++) {
            if (map.containsKey(l2[i]) && map.get(l2[i]) + i <= min) {
                if (map.get(l2[i]) + i == min) {
                    sb.append(l2[i]).append(",");
                    continue;
                }
                head = sb.length();
                sb.append(l2[i]).append(",");
                min = map.get(l2[i]) + i;
            }
        }
        return sb.substring(head, sb.length() - 1).split(",");
    }

    public String[] findRestaurantRefer(String[] list1, String[] list2) {
        // HashMap无序
        Map<String, Integer> map = new HashMap<>(list1.length * 2);
        // 存储索引和
        int minIndex = -1;
        // 临时存储值的
        List<String> list = new ArrayList<>();

        for (int i = 0; i < list1.length; i++) {
            map.put(list1[i], i);
        }

        for (int i = 0; i < list2.length; i++) {
            if (map.containsKey(list2[i])) {
                // 表示第一次遇到公共值
                if (minIndex == -1) {
                    minIndex = map.get(list2[i]) + i;
                    // 第一次公共值结束
                } else {
                    if (minIndex == map.get(list2[i]) + i) {
                        minIndex = map.get(list2[i]) + i;
                    } else if (minIndex > map.get(list2[i]) + i) {
                        list.clear();
                        minIndex = map.get(list2[i]) + i;
                    } else {
                        break;
                    }
                }
                // 包含都要添加的，如果最小下标更改，上面执行了list.clear()方法就好
                list.add(list2[i]);
            }
        }
        // 最终结果
        String[] strings = new String[list.size()];
        for (int i = 0; i < list.size(); i++) {
            strings[i] = list.get(i);
        }
        return strings;
    }

    public static void main(String[] args) {
        String[] s1 = {"S", "TEXP", "BK", "KFC"};
        String[] s2 = {"KFC", "BK", "S"};
        Solution100 solution100 = new Solution100();
        System.out.println(new Gson().toJson(solution100.findRestaurant(s1, s2)));
    }
}
